import LSN2Vec.lsNet2VecClass as ls2Vec
from LSN2Vec import graph_util
from sklearn.model_selection import train_test_split
import numpy as np
if __name__ == '__main__':
    data = np.matrix(np.loadtxt('../vecedNodes.csv',delimiter=","))

    edge_set = np.loadtxt('../edgeset.csv',delimiter=",",dtype='int')
    ab_set = np.loadtxt('../abset.csv',delimiter=",", dtype='int')
    y = [1] * len(edge_set)
    _, test_set, _, _ = train_test_split(edge_set, y, test_size=0.2)
    auc_result = graph_util.cal_auc(test_set, ab_set, data)
    print(auc_result)
